package model

import (
	"gin/common"
	Mysql "gin/database"
	"github.com/Gre-Z/common/jtime"
	"strings"
)

type SentenceNerToken struct {
	Id          uint64           `json:"id" gorm:"column:id;primaryKey;type:int;autoIncrement"`
	BeginOffset uint64           `json:"begin_offset" gorm:"column:begin_offset"`
	Word        string           `json:"word" gorm:"column:word"`
	Length      uint64           `json:"length" gorm:"column:length"`
	Type        string           `json:"type" gorm:"column:type"`
	SentenceId  uint64           `json:"sentence_id" gorm:"column:sentence_id"`
	Created     jtime.TstampTime `json:"created" gorm:"column:created;autoCreateTime"`
}

// 设置表名
func (SentenceNerToken) TableName() string {
	return "sentence_ner_token"
}

func (this *SentenceNerToken) Insert() (id uint64, err error) {
	result := Mysql.DB.Create(&this)
	id = this.Id
	if result.Error != nil {
		err = result.Error
		return
	}
	return
}

func (this *SentenceNerToken) Find(Param common.SelectQueryParam) (resData SentenceNerToken, err error) {

	dictOne := new(SentenceNerToken)
	db := Mysql.DB

	// 组装过滤条件
	if Param.Filter.Filter != nil {
		for key, val := range Param.Filter.Filter {
			vals := val.(map[string]interface{})
			if strings.Contains(key, "bw_") {
				db = db.Where(vals["condition"], vals["val1"], vals["val2"])
			} else {
				db = db.Where(vals["condition"], vals["val"])
			}
		}
	} else {
		db = db.Where("?", "1=1")
	}

	// 应用分页
	db = db.Select(Param.Fields)

	result := db.First(&dictOne)
	if result.Error != nil {
		err = result.Error
	}

	return *dictOne, err
}

func (this *SentenceNerToken) Select(Param common.SelectQueryParam) (total int64, resData []SentenceNerToken, err error) {

	models := make([]SentenceNerToken, 0)
	db := Mysql.DB

	// 组装过滤条件
	if Param.Filter.Filter != nil {
		for key, val := range Param.Filter.Filter {
			vals := val.(map[string]interface{})
			if strings.Contains(key, "bw_") {
				db = db.Where(vals["condition"], vals["val1"], vals["val2"])
			} else {
				db = db.Where(vals["condition"], vals["val"])
			}
		}
	} else {
		db = db.Where("?", "1=1")
	}

	// 获取总条数
	db.Model(&SentenceNerToken{}).Count(&total)

	// 查找所有数据

	// 应用分页
	db = db.Select(Param.Fields).Limit(Param.Limit).Offset(Param.Offset)

	// 应用排序参数
	if Param.Order != "" {
		db = db.Order(Param.Order)
	}

	result := db.Find(&models)
	if result.Error != nil {
		err = result.Error
		return
	}

	return total, models, err
}

func (this *SentenceNerToken) FindById(id uint64) (resData SentenceNerToken, err error) {

	db := Mysql.DB

	db = db.Where("id = ?", id)

	// 应用分页
	db = db.Select("*")

	result := db.First(&this)
	if result.Error != nil {
		err = result.Error
		return *this, err
	}

	return *this, err
}

// 更新dict
func (this *SentenceNerToken) Update(sentenceNerToken SentenceNerToken) (SentenceNerToken, error) {
	db := Mysql.DB
	result := db.Updates(&sentenceNerToken)
	if result.Error != nil {
		return sentenceNerToken, result.Error
	}
	return this.FindById(sentenceNerToken.Id)
}

func (this *SentenceNerToken) Delete(dict SentenceNerToken) {
	Mysql.DB.Delete(&dict)
}

// 批量删除
func (this *SentenceNerToken) DeleteAll(where string) {
	Mysql.DB.Where(where).Delete(&SentenceNerToken{})
	return
}

// 批量插入
func (this *SentenceNerToken) InsertAll(sentenceNerTokens *[]SentenceNerToken) *[]SentenceNerToken {
	Mysql.DB.Create(&*sentenceNerTokens)
	return sentenceNerTokens
}

// 批量删除
func (this *SentenceNerToken) DeleteByFilter(param *common.Filter) {
	db := Mysql.DB

	if param.Filter == nil {
		return
	}
	// 组装过滤条件

	for key, val := range param.Filter {
		vals := val.(map[string]interface{})
		if strings.Contains(key, "bw_") {
			db = db.Where(vals["condition"], vals["val1"], vals["val2"])
		} else {
			db = db.Where(vals["condition"], vals["val"])
		}
	}
	db.Delete(SentenceNerToken{})

}
